from modules import *
import random
from utils.tools import 暂无数据,随机获取列表

class 行政_短信通知(Page_Object):
    def __init__(self, page: Page):
        super().__init__(page)
        self.手机号查询 = self.page.get_by_placeholder("手机号查询")

        self.查询 = self.page.get_by_text("查询")
        self.重置 = self.page.get_by_text("重置")

        # self.暂无数据 = self.page.locator(".el-table__empty-text")
        self.搜索结果第一列 = self.page.locator(".el-table__body-wrapper").locator("tbody").locator("tr").nth(0).locator("td").nth(0)


    def navigate(self):
        self.跳转("/approval/message")

    def search_搜索(self, 手机号查询):
        # 断言文本为“查询”的元素可见（自动等待，最多5秒）
        self.手机号查询.fill(手机号查询)
        self.查询.click()
        expect(self.搜索结果第一列).to_be_visible(timeout=2000)
        self.重置.click()
        self.page.wait_for_timeout(1000)

        # 区域选择街道 = self.区域选择街道
        # count = 区域选择街道.count()
        # if count > 0:
        #     # 随机选择一个索引
        #     random_index = random.randint(0, count - 1)
        #     # 点击对应的 li
        #     区域选择街道.nth(random_index).click()
